package xin.fsqr520.project.util;

import java.util.Random;

/**
 * 验证码生成工具类
 */
public class VerificationCodeUtil {

    private static final Random RANDOM = new Random();
    private static final int CODE_LENGTH = 6; // 验证码长度
    
    /**
     * 生成6位数字验证码
     * @return 6位数字验证码字符串
     */
    public static String nextCode() {
        // 生成100000-999999之间的随机数
        int code = RANDOM.nextInt(900000) + 100000;
        return String.valueOf(code);
    }
    
    /**
     * 生成指定位数的数字验证码
     * @param length 验证码长度
     * @return 数字验证码字符串
     */
    public static String nextCode(int length) {
        if (length <= 0) {
            throw new IllegalArgumentException("验证码长度必须大于0");
        }
        
        // 计算范围
        int min = (int) Math.pow(10, length - 1);
        int max = (int) Math.pow(10, length) - 1;
        
        int code = RANDOM.nextInt(max - min + 1) + min;
        return String.valueOf(code);
    }
}